package com.yeb.server.task;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.yeb.server.pojo.Employee;
import com.yeb.server.pojo.MailConstants;
import com.yeb.server.pojo.MailLog;
import com.yeb.server.service.IEmployeeService;
import com.yeb.server.service.IMailLogService;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Component;

import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.List;

/**
 *  邮件发送定时任务
 * @Author liushizhong
 * @Date 2021/12/31 14:11
 * @Version 1.0
 */

@Component
public class MailTask {

    @Autowired
    private IMailLogService mailLogService;
    @Autowired
    private IEmployeeService employeeService;
    @Autowired
    private RabbitTemplate rabbitTemplate;

    /**
     * 每10秒发送一次
     */
    @Scheduled(cron = "0/10 * * * * ?")
    public void mailTask(){
        // 获取正在发送的邮件并且当前时间大于重试时间
        List<MailLog> list = mailLogService.list(new QueryWrapper<MailLog>().eq("status", 0)
                .lt("tryTime", LocalDateTime.now()));
        list.forEach(mailLog -> {
            /**
             * 重试次数超过3次，更新状态为发送失败
             */
            if(3 <= mailLog.getCount()){
                mailLogService.update(new UpdateWrapper<MailLog>().set("status",2)
                        .eq("msgId",mailLog.getMsgId()));
            }else {
                // 重试发送邮件，更新相关状态
                mailLogService.update(new UpdateWrapper<MailLog>().set("count",mailLog.getCount() + 1)
                        .set("updateTime", LocalDateTime.now()).set("tryTime",LocalDateTime.now()
                                .plusMinutes(MailConstants.MSG_TIMEOUT)).eq("msgId",mailLog.getMsgId()));
                // 获取员工
                Employee employee = employeeService.getEmployee(mailLog.getEid()).get(0);
                // 重发邮件
                rabbitTemplate.convertAndSend(MailConstants.MAIL_EXCHANGE_NAME,MailConstants.MAIL_ROUTING_KEY_NAME,employee,
                        new CorrelationData(mailLog.getMsgId()));
            }
        });
    }

}
